config_dict = {
    "data_path": "/path/to/data",
    "loss_type": "cross_entropy",
    "optimizer_type": "adam",
    "scheduler_type": "step",
    "epochs": 10,
    "batch_size": 32,
}